#include"conv.h"
#include"data_transfer.h"

At op(At x,Wt y);

void conv2d(volatile At* in1,volatile At* in2,const Wt weight[Tm][Tn][K][K],Ot out[Tm][MAX_LEN*MAX_LEN],int fsize,int ch_in);

void add_partial_sum(Ot obuffer[Tm][MAX_LEN*MAX_LEN],Ot pbuffer[Tm][MAX_LEN*MAX_LEN],int ofsize);

void compute_one_block(volatile At* in1,volatile At* in2,const Wt weight[M][N][K][K],Ot obuffer[Tm][MAX_LEN*MAX_LEN],int fsize,int ch_in,int ch_out);
